import streamlit as st
import requests

# 设置 LangServe API 的地址
LANG_SERVE_URL = "http://localhost:8000"  # 根据实际部署的 LangServe 服务地址进行修改


# Streamlit 应用标题
st.title("LangServe + Streamlit 交互界面")

# 输入框
user_input = st.text_input("请输入你的问题或指令：", placeholder="例如：你好，你是谁？")

# 提交按钮
if st.button("提交"):
    if user_input:
        # 调用 LangServe API
        response = requests.post(
            f"{LANG_SERVE_URL}/invoke",  # LangServe 的默认端点
            json={"input": user_input}
        )

        # 检查响应状态
        if response.status_code == 200:
            # 显示 API 返回的结果
            result = response.json().get("output", "未获取到结果")
            st.success(f"结果：{result}")
        else:
            st.error(f"请求失败，状态码：{response.status_code}")
    else:
        st.warning("请输入内容后再提交！")